import sys
import os

# 添加项目根目录到Python路径
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from src.eng_translate_far import *

# 执行训练
if __name__ == "__main__":
    print("正在初始化数据...")
    print(f"英文词汇表大小: {english_word_n}")
    print(f"法文词汇表大小: {france_word_n}")
    print(f"训练样本数量: {len(my_pairs)}")
    print(f"使用设备: {device}")
    print("\n开始训练Transformer模型（测试模式）...")

    transformer_losses = use_transformer(test_mode=True)
    print("\nTransformer模型测试训练完成！")

    # use_transformer()  # 这行是完整训练的调用方式
